package br.mil.cpp.modelo;

import java.util.List;

import javax.swing.table.AbstractTableModel;

public class ModalidadeTableModel extends AbstractTableModel{
	
	private List<Modalidade> modalidades;

	public ModalidadeTableModel(List<Modalidade> modalidades) {
		this.modalidades = modalidades;
	}
	
	@Override
	public int getColumnCount() {
		return 3;
	}

	@Override
	public int getRowCount() {
		return modalidades.size();
	}

	@Override
	public Object getValueAt(int linha, int coluna) {
		Modalidade modalidade = modalidades.get(linha);
		
		switch (coluna) {
		case 0:
			return modalidade.getModalidade();
		case 1:
			return modalidade.getDias();
		case 2:
			return modalidade.getProfessor();

		}
		
		return "";
	}
	
	@Override
	public String getColumnName(int coluna) {
		switch (coluna) {
		case 0:
			return "Modalidade";
		case 1:
			return "Dias";
		case 2:
			return "Professor";
		}
		
		return "";
	}

	public void adicionar(Modalidade modalidade) {
		this.modalidades.add(modalidade);
	}
	
	public void excluir(Modalidade modalidade) {
		this.modalidades.remove(modalidade);
	}
}
